quiz python code